Update benchmark.py for system info