Add if conditionals