I usually add one or more reset functions, and then I can customize whatever state I want to return to.