Update categorical.py