fixed correlation refactor a bit so now it should be more extensible when we add support for data from file