input: Write a python function to generate code for a given prompt based on the specified programming language.
  assert generate_code('python', 'print hello world') == 'print("hello world")'
  assert generate_code('javascript', 'print hello world') == 'console.log("hello world");'
  assert generate_code('java', 'print hello world') == 'System.out.println("hello world");'
output: def generate_code(language, prompt):
      if language == 'python':
          return f'print("{prompt.split(" ")[1]} {prompt.split(" ")[2]}")'
      elif language == 'javascript':
          return f'console.log("{prompt.split(" ")[1]} {prompt.split(" ")[2]}");'
      elif language == 'java':
          return f'System.out.println("{prompt.split(" ")[1]} {prompt.split(" ")[2]}");'