Started to add test edit functionality