Add multiline check for if