Cleaned up benchmark script somewhat.