Add simple unit tests