Training the agent visually