Yes, the upgraded version will be much faster for these kinds of problem where what is required is just the moments of the solution. Propagating the entire probability distribution just to compute the moments is a very expensive way to do the calculation.
For plotting see Plots.jl and Makie.jl