Adding in the hw2 assn.