Used switch statement instead of if-else

[connect-bilal]

Using a switch statement makes the code more readable and easier to manage when dealing with different cases based on the type.

[nao-pon]

@bilalkhan13 I'm sorry to be late. Merge.