HW1
Moved to python notebook