Added more test for the editing functionality