Repo for data science course project