input: Write a function to edit the given code to make it more efficient. The function should use list comprehension to simplify the for loop and if condition. If the input list is empty, the function should return an empty list.
  assert improve_code([1, 2, 3, 4])==[3, 4]
  assert improve_code([])==[]
  assert improve_code([1, 2, 2, 2])==[]
  assert improve_code([3, 4, 5, 6])==[3, 4, 5, 6]
  assert improve_code([2, 2, 3, 2])==[3]
output: def improve_code(lst):
      return [x for x in lst if x > 2] if lst else []