Adding HW2 python notebook