In this case for me it's more readable to use the `switch` instead of `if/else`, do you mind if we keep this structure? Is there any reason for the change? Thanks!