made some changes to notebook and endgpraph.py