Added class definitions in RobotMain