My first project for the data science course