Added test notebook