add a couple of unit tests