Finished a unit test for all the basic functionality