robot DONE working code