cleaned up robot.py