Add functionality to save/discard/cancel