Assignment 4 Classification Solution