CS231 Assignment 6