Update install script with new model

Fixes #209