python populate dictionary with for loop