Update check_mem.py