Add support for profiling