Update code after profiling