I modify some codes.

I modify some codes.