modified python scripts for benchmark