repository for data science course project