CS 221: Completed assignment 6