Update motion_model.py