Update memory.py