Add country field to Address model