Plots module, cleanup imports on test