additions to edit UI