Add memory checking
Enable memory collection