Fix #1 : reset diagram after clicking "Back to Edit Mode"