Made it so that we can see the top 6 correlated classes