Add reset feature etc