Allow different cosmology to be specified

I noticed that the cosmology is hardcoded to be Planck15, it would be good to allow the user to specify another cosmology.

It would probably be fairly straightforward to use the bilby cosmology functionality, https://git.ligo.org/lscsoft/bilby/blob/master/bilby/gw/cosmology.py.