Update sorting-hat.py