modified python_benchmark/draw_time_dim.py