public function writeDefinition(array $definition) { if (isset($definition['is_main_class']) && $definition['is_main_class'] === true) { $definition['inheritance']['extends'] = $this->_getBaseClassName($definition['className']); } return parent::writeDefinition($definition); }