Add support for plotting 3d and for plotting with color