Add more reset button functionality