Replace class generator with the submodule created from it