Added basic correlation calculation