Added benchmark to test.py