Added ability to save country