Updated tool to have a reset feature