Add setLegend() method to redefine legend scale