Robot.py minor main function changes