Code for robot competition