Homework 2 Assignment for CS335