if...else -> case...when

Signed-off-by: Timothy Gu <timothygu99@gmail.com>