Add a way to remove update message if needed