Minor refactoring of host.JITModule for consistency
Documentation updates and clarifying comments