Update py_mem.py