Created some unit tests to verify the code