Added switch, instead if-else