Adding bench.py