HW1 ipython notebook