Work on ex36.py