Cleaned up cosmology section