Add machine learning to the project