added a couple more unit tests