Adding some basic unit tests.