[python] Populating a dictionary using for loops (python)