Updated notebook so that it works with new directory configuration.