updated correlation function test notebook