Added unit testing

HW3 complete!