Modified notebook to run in test_dir in workflow demo.