Add some logic.