changed the retriever class