Adds some unit tests.