With this change the argument only allows one of the 3 choices
It also takes a default value in case the arg is not given