small changes to legend