Adding simple unit tests