Cosmology script added.

Copied the cosmology script for figure 8.5 from Astro-ML in preparation
for next stage of research project.