Flush output after prompt. Fixes #2