add correlation calculation