Add CodeGenerator dependency