Add code for instrumenting the heap