Consider taking `sqrt` earlier in plotting script

[bryates]

Should we use np.sqrt here, or do we use the squared values directly elsewhere?

Originally posted by @bryates in #492 (comment)