add model for country