Overload correlation functions. Need to test.