Just a note that we should replace all occurrences of .plot_psd() with .compute_psd().plot().