Create Done button to return to previous menu after editing, fixes #3