Use if else instead of case because it's more obvious