Add support for plotting categorical data

Code still needs cleanup and support for a legend.