Updated notebooks to work with new code package.