Adding a flag to separate initial and final trials