Add related command to benchmark.py