CSC3022H - Machine Learning assignment 2