add a couple more unit tests