Nevermind this breaks it, fix get_norm with a simple square integrator maybe