Adding simple unit test