fix list comprehension, add zip