Added in model loading code.