Updated benchmark.py after adding usage/docstrings etc.