More work on plot_population.py
.