Update memory.py

Updating labels