Use Swift's type system to restrict values

Good catch - and I wonder if this is better suited for OPs case than Enum