Fix some missing variables and remove block kwarg from plt.show
Update notebook