Added simple unit test