Add if then posibilities

Adding extra logic into the test cases depending on the value of a variable.