Minor change to use enumeration
Fix for #56 - CLOSED