updated notebook with new figures including studies prediction