Implement better permutation test

[TimotheeMathieu]

Use this: https://arxiv.org/abs/2204.13581