updated hw3 ipython notebook