Update memprof.py