Add Python version of the benchmark