Fix indentation
Refactor and simplify tests
Add 2D Welford benchmark