Repo for data science final project