hw4

Final version of hw4, but need to add unit test file.