Add bench.py