Added code to find more variable and plot 2d