A little change.
Finish the logic.