Robot game with Pygame