refactored doc and nn Modules.