input: Write a function to modify the given Python code such that it raises a TypeError exception if the input value is not an integer. Replace the original `square_root` function with the modified version that performs this check. The function should return the square root of the input if it is an integer.
  assert square_root(4)==2.0
  assert square_root(9)==3.0
  assert square_root('4')==TypeError('Input must be an integer')
  assert square_root(2.5)==TypeError('Input must be an integer')
  assert square_root(16)==4.0
output: def square_root(n):
      if not isinstance(n, int):
          raise TypeError('Input must be an integer')
      return n ** 0.5