Add memory usage to integration tests [UPDATED]