Yup, that's a good approach, and probably add a condition on when to show that button.