Tidying up the robots code :)