added new functionallity to generator