adding a few more unit tests