add output stream draining