Allow models to be specified more pythonically - with type hints, perhaps use pydantic for type checking #147