Add Heap tracking

[juliexiong]

Any tracking from Heap would give us insight into how we can improve Booster.