Update test notebook