Can specify a model size, closes #12