update project4.py
Merge pull request #2 from sirr/patch-2

update project4.py