Working on HTML edit functionality