added 2 simple unit tests