Add the functionality to hide the toast