Modifying finite difference method for state