Machine learning project for the data science track