Create hw4.py