[LK] Pass model into init method and do initialization there