Getting closer to implicit 2d scheme, but still something wrong