Add profiling utils