add bench.py