Bug and nit fix CountingSort.py