Added classes for loading the model.