Update python.py

Untested, just trying to get you to think a bit differently