Change if/else on switch