For the final project of Data Science