Remove function for generator statement