Updating HW1 python notebook