Create hw1.py

First homework assignment