update pe classification notebook