Ability to use custom models