Add tests for more profiler functions