Adding a couple unit tests