Add a few simple unit tests