Added time profiling to test file