Add in class assignment 2
Finish hw3