Adding profiler support