swapping in list comprehension