Updated reset functionality