input: Write a python function to get the capital city of a given country from a dictionary.
  assert get_capital("UK") == "London"
  assert get_capital("France") == "Paris"
  assert get_capital("Spain") == "Madrid"
output: def get_capital(country):
      capitals = {
          "UK": "London",
          "France": "Paris",
          "Spain": "Madrid",
          # Add more countries and capitals as needed
      }
      return capitals.get(country, "Capital not found")