Refactor how the generator.shell module is added.