input: Write a function to remove all occurrences of a given element from a given array. The function should return the array after the removals.
  def remove_element(arr, element):
      arr = [x for x in arr if x != element]
      return arr
  assert remove_element([2, 4, 5, 7, 4, 2], 4) == [2, 5, 7, 2]
  assert remove_element([1, 2, 3, 4, 5], 3) == [1, 2, 4, 5]
  assert remove_element([10, 20, 30, 10, 40], 10) == [20, 30, 40]
  assert remove_element([], 5) == []
output: def remove_element(arr, element):
      arr = [x for x in arr if x != element]
      return arr