Added edit function (ongoing)