Modified the GUI of the edit again