diff --git a/ce_ranker.py b/ce_ranker.py
index 49a9a04..e848909 100644
--- a/ce_ranker.py
+++ b/ce_ranker.py
@@ -59,10 +59,8 @@ class CERankerAgent(PrecompiledAgent):
             *[self._score_document(query, doc) for doc in documents]
         )
 
-        # Sort by the score (descending). If score is True, it comes first.
         scored_results.sort(key=lambda x: x[1], reverse=True)
 
-        # Return a list of documents (not joined as a string), up to k results
         return [doc for doc, score in scored_results[:k]]
 
     def __call__(self, query: str, k: int | None = None) -> list[str]: