Assignment 5: Classification Solution