Wave function fluxonium

The method wave_function in fluxonium works only if the Fock basis is selected. 
So it would not work if we change the basis as we do after diagonalize. In general, 
I think the change of basis can lead to issues of this kind