Adding the code to move the robot based on regression data