ENH: added categories