updated notebook with code