Making an image plot or heat map