Added the edit option to the method. Changed the initial setting for the done button when editing.