refactored bench.py and made it more versatile