Added Colorbar and adjusted dimensions