Update map_us.py to include subplots