Add profiler support