Add colored rects as legend for small states
Change small state label orientation