Python code related to robot simulation