added ability to load model too