add conditional statements