updated tf notebook