Merging in Claudia's LM Retrieval code