Adding usability request: done button after successfully saving