Changed main method to robot class