Code for Machine Learning Homework