Added code to python.py

Yes.