add else if, ternary