Added type to model