Test stuff in plots
Correction and small addition