Edit on click mostly implemented