fun foo(a: Int) = if (a > 1) { (a + 1) * (a - 1) } else { a * (a + 1) }