Repo for data science assignment