Add a reset function