Updated the ModuleList with KeyOutput and StateSaver