Updated code to match the new robot design.