Added two simple unit tests