updated notebook with colorful plots