fix colorbar=True parameter in 2D plotting functions