Switch instead of if