Adding conditional statements