Nit: what about having an enum?