Assignment 2 - ML - CS 595