Add getModel method to allow greater flexibility