input: Write a function to remove all occurrences of a given element from a given array and return the modified array.
  assert remove_element([2, 4, 5, 7, 4, 2], 4) == [2, 5, 7, 2]
  assert remove_element([3, 5, 5, 7, 8, 5], 5) == [3, 7, 8]
  assert remove_element([2, 4, 5, 4, 7, 2], 2) == [4, 5, 4, 7]
  assert remove_element([], 4) == []
  assert remove_element([2, 2, 2, 2], 2) == []
output: def remove_element(arr, element):
      return [x for x in arr if x != element]