using zip function in python