notebook changes
fix nan problem