Added simple unit testing.