updated notebook to reflect changes in dir structure