Assignment 3 (A3) Machine Learning Solution