Adding in support for plotting the solution and covariance matrix