Add reset functionality