Update notebook to reflect the most recent changes

[andersy005]

Description of proposed changes

Fixes #

[codecov-commenter]

Codecov Report

Merging #114 (21c86d1) into main (77c470b) will not change coverage. The diff coverage is n/a.

@@            Coverage Diff            @@
##              main      #114   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files            5         5           
  Lines          192       192           
  Branches        35        35           
=========================================
  Hits           192       192           

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.