Add callback function to reset