Adding some unit tests.