add some logic