Modifications to plot_population.py