Counting Sort in Python