python list comprehension add two elements