Added editing functionality