Added wave function