“Write down the completeness relation for the general 3-state case.”
After the completeness relation has been presented in the most familiar context of the spin-$\dfrac{1}{2}$ system, practice converting the same concept to a system with more states is useful for solidifying the general concept of the completeness relation. This small white board question also helps the instructor check who understands the general concept of the completeness relation and who still needs assistance.