updated for different output format from new cosmolopy