do we really need all the classes `RadialState`, `Wavefunction` and `Grid`, or can this be simplified