add time import back to python_benchmarks.py