Add initModel() function