Add the reset ability