Data Science Machine Learning Assignment