Added simple unit tests