Update proj2.py