aded bench.py and some comments
make it faster