Adding hw1 python notebook