Add new methods to generator