Add reset button functionality