create list of lists using zip in python