Added model type to props