New module names in docs

Follow up to #285, by fixing the module names in the docs.