Fix up example notebooks

Hopefuly this won't have tons of commits