Simple drive code for a robot