Is there any way to make llm's better at retaining memory?