Switched to facetgrid for category