if else & for loop