updated notebook with code1