Remove dependency on PyPlot