Building a mini search engine from scratch in Python