before new evolution eqs
implemented new version of evolution equations. Not stable yet. Will try einsum from this point