Added the enum class