Added bench.py for benchmarking