added the two tests that are in the assignment