ML Assignment 1