Add reset function