Slight modification to legend