updated notebook for figrue 6