Machine Learning Assignment 2