Minor doc fixes
Use setup method to ensure modules are loaded properly
Fix incorrect documentation