Use np corrcoef