Correct doc formatting

Include the new module in autodoc