Better use of enums