Could you use the enum?