Pythonically using dictionaries