repo for data science course project