Test Additions

Created tests for the use cases.