Initial commit of machine learning project