Optimization for larger models