A little polish.

- Fixed module name in Haddock comment.
- Marked main payload as -XSafe.
- Clarified some comments.

- Added more tests, copied from `units`.