Assignment 3, multi-class classifier