Короткая запись функции в Kotlin

Если функция только возвращает результат одного выражения, тело можно не оборачивать в фигурные скобки. После типа результата ставят = и выражение:

fun func(num: Int): Int = num * num

fun main() {
	println(func(3)) // 9
}

Сумма двух параметров записывается так же:

fun func(num1: Int, num2: Int): Int = num1 + num2

fun main() {
	println(func(1, 2)) // 3
}

Короткая форма подходит и для условия, которое само дает Boolean:

fun func(a: Int, b: Int): Boolean = a > b

fun main() {
	println(func(3, 2)) // true
}

Запишите функцию func с параметром целого числа в короткой форме. Она должна возвращать квадрат числа.

Дана функция:

fun func(a: Int, b: Int): Boolean {
	if (a == b) {
		return true
	} else {
		return false
	}
}

Перепишите ее в короткой форме.