Add testing for do_edit