Make Flush work without requiring curly braces

@pranavkm - Could you please take a look at the PR?