Updates to scikit learn notebook, homework 1