Clarified API for the create_model function,
Also added a test!