Add button that can save the result