Remove top-k and add weight threshold