Fix imports for standalone memory benchmark scripts
Merge pull request #3 from aniketschneider/memory-benchmark-imports

Fix imports for standalone memory benchmark scripts