Optimization for duplicate code removal