Added the logic.