Added categorical information
Added a legend for the categories