Ensuring no changes in original code flow