Noticed missing model params.  Added them back in