add additional unit tests