Update categoryDisplay.py