edit functionality changes