Update ps_mem.py