Added initial profiling code