Improve ocaml module documentation

Add some documentation about environment assumptions, and correct some
grammar issues.