added additional unit tests