I would like a switch better than if-else for this.