Change model initialization