load model function added