Adding wavefunction class, other files