Added legend
better legend styling