Added hw1.py

First assignment