Crank Nicolson method works