* add some tests * Fix topk bug * Fix bug in test * Add remaining tests * Remvoe debug code * Delete redundant code * Remove redundant code