CS 221 Assignment 5