CS193p Assignment 5 Code