Update easy_physics.py