Working with maps in Python