Make legend a bit better