adding first version of hw1
add hw1.py