add conditional statement