Add reset button and reset functionality