Add a couple of more unit tests