Add await-flag function.