Replace deprecated param_dim with n_models
Merge pull request #109 from larrybradley/model_update

Replace deprecated param_dim with n_models