Update topk.scala