Updates for profiling code