improvements for diversification plot