Fix #3: Add support for enum type