added correlation visualization