Added scaling for larger models