square plot, created a colourmap but not properly working yet