Reorganised the robot draw function