Add test for profiling