time evolution of wave functions work again, current still wrong