Create gravity.py