Update topK.py