made the class names in model_loading.py compliant with the ones used in models.py