repo for data science assignment