Added model for country