Add simple profiling to Driver