Update the if statement to support(ish) block