Intro to Data Science Assignment 1