Add reset feature