Add legend
Correct legend