Clean out non-useful wave function