Updated cor function