Update gravity-well.py

Final commit.