add unit tests and fix wording