Added ability to specify model - closes #1