add some logic.