Fixing legend issue.
Fixing legend issue. Not really there yet.