Add flush method