Add _flush method