Final commit of the machine learning project