Added a function to edit labels