clean up generator style