With plotting
Revert "minor updates"

This reverts commit 9a2dc99145f5907d3ef36c6f4bfb3d6a83249ac7.