Add profiling (and flag for it)