fleshed out the purity plotting notebook even more