Added setModel method