intro to data science - assignment 5