1 files changed, 5 insertions, 7 deletions
diff --git a/Swiften/Elements/DiscoInfo.h b/Swiften/Elements/DiscoInfo.h
index b73165e..c5c9e1c 100644
--- a/Swiften/Elements/DiscoInfo.h
+++ b/Swiften/Elements/DiscoInfo.h
@@ -7,12 +7,10 @@
 #pragma once
 
 #include <vector>
-#include <algorithm>
-
-#include "Swiften/Elements/Payload.h"
 #include <string>
 
-#include "Swiften/Elements/Form.h"
+#include <Swiften/Elements/Payload.h>
+#include <Swiften/Elements/Form.h>
 
 namespace Swift {
 	class DiscoInfo : public Payload {
@@ -23,6 +21,8 @@ namespace Swift {
 			static const std::string SecurityLabelsFeature;
 			static const std::string SecurityLabelsCatalogFeature;
 			static const std::string JabberSearchFeature;
+			static const std::string CommandsFeature;
+			static const std::string MessageCorrectionFeature;
 
 			class Identity {
 				public:
@@ -82,9 +82,7 @@ namespace Swift {
 				features_.push_back(feature);
 			}
 
-			bool hasFeature(const std::string& feature) const {
-				return std::find(features_.begin(), features_.end(), feature) != features_.end();
-			}
+			bool hasFeature(const std::string& feature) const;
 
 			void addExtension(Form::ref form) {
 				extensions_.push_back(form);