fixing robot.py