Adding robot code