Top K Update

Updated TopK code.