Move change of label to controller