working on editing functionality