almost working
trying to implement top k now