Added edit functionality