Cleaned up Scikit-learn notebook