Rearrange if-else