Legend display tweaks

Thanks!