Update customSort.py