Add profiling code
Remove profiling code