Minor fixes to legend layout