Machine Learning Final Project code