adding some logic