Update control flow statements