Added init model in Initialization