Update plotting. Pull in submodule