Add import to extract_maps.py for print_function