Add reset functionality and button