Use switch
Refactor