Add memory watching