matplotlib changes in the legend