Design a Python program which takes two lists and returns their intersection.
list1 = [1, 2, 3, 4, 5, 6]
list2 = [3, 6, 8, 9, 10]
def intersection(list1, list2): 
    return list(set(list1) & set(list2)) 

# Output: [3, 6]