Updated movement.py