Repo for Data Science course (Final project)