Small refactor
Another poly function and tests
Add moment_for_box