Add option to draw legend for categorical data