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