Model loading added