Fix documentation and switch enum to NS_ENUM