Assignment 1 Introduction to Python & sklearn Solution