First assignment for data science class