input: Write a python function to format a given string with the correct indentation levels.
  assert format_text("This is\na string\nwith different lines:") == "This is\n  a string\n  with different lines:\n"
  assert format_text("def func():\nreturn 5") == "def func():\n  return 5\n"
  assert format_text("if condition:\ncode\nelif another_condition:\ncode2") == "if condition:\n  code\nelif another_condition:\n  code2\n"
output: def format_text(text):
      lines = text.split('\n')
      formatted_lines = [lines[0]]
      for line in lines[1:]:
          formatted_lines.append('  ' + line)
      return '\n'.join(formatted_lines) + '\n'