Adding HW1 python notebook