def array_diff(a, b):
    return a if not b else [x for x in a if x != b[0]]

def array_diff(a, b):
    return [x for x in a if x not in set(b)]

Change code because failed the random test
Change code because failed the random test
