Working on retriever.py