Fix benchmark script.
Merge pull request #17 from siong1987/benchmark

Fix benchmark script.