Can we use an enum for this?