More work on representing country. Still need to fix prompt.