better enum
Merge pull request #2 from sehraf/master

better enum