Added State enum class