Adding the label functionality