Code created for Machine Learning course