Nested for loops in python list comprehension