Updates to notebook example.