diff --git a/.vscode/settings.json b/.vscode/settings.json
index 51284b6c40da19a240724911b60bc78704d18fa4..3fc0fa60242927e742bf86bd80aee39babd6870c 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -6,5 +6,6 @@
 	"editor.codeActionsOnSave": {
 		"source.fixAll.eslint": true
 	},
-	"languageServer.wdChecks": false
+	"languageServer.wdChecks": false,
+	"languageServer.probHome": "/home/sebastian/prob_prolog/probcli.sh"
 }
\ No newline at end of file
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 161d47d3e09313b45b64688438ca3d77ed9bd8e3..a62e17d4ec949380081849c77b7b5d3c45fe8970 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -27,4 +27,11 @@
 
 ## [0.1.1]
 
-- fixed issue where path was semi hard coded
\ No newline at end of file
+- fixed issue where path was semi hard coded
+
+
+
+## [0.2.0]
+
+- recognizes correct file ending (.mch; .def; .imp; .ref; .sys)
+- no longer activates extension on normal text documents
\ No newline at end of file
diff --git a/README.md b/README.md
index ee0ce5fa897a03351d892f5cb8de51c794bcd5d1..9e22d8c8dc54de2cc0c9a49b8850376fdeb31fce 100644
--- a/README.md
+++ b/README.md
@@ -68,4 +68,10 @@ Please note that user settings overwrite workspace settings.
 
 ### 0.1.1
 
-- fixed issue where path was semi hard coded
\ No newline at end of file
+- fixed issue where path was semi hard coded
+
+
+### 0.2.0
+
+- recognizes correct file ending (.mch; .def; .imp; .ref; .sys)
+- no longer activates extension on normal text documents
\ No newline at end of file
diff --git a/client/src/extension.ts b/client/src/extension.ts
index 69ecdeab7817e4f317c3e206751d041b5f220bed..32890ee849e2941b7d5e1497de1bda9a681ca7cd 100644
--- a/client/src/extension.ts
+++ b/client/src/extension.ts
@@ -38,8 +38,8 @@ export function activate(context: ExtensionContext) {
 
 	// Options to control the language client
 	let clientOptions: LanguageClientOptions = {
-		// Register the server for plain text documents
-		documentSelector: [{ scheme: 'file', language: 'plaintext' }],
+		// Register the server for B, EventB documents
+		documentSelector: [{ scheme: 'file', language: 'B' }, { scheme: 'file', language: 'EventB' }],
 		synchronize: {
 			// Notify the server about file changes to '.clientrc files contained in the workspace
 			fileEvents: workspace.createFileSystemWatcher('**/.clientrc')
diff --git a/package.json b/package.json
index 8081649ebf94026cae61c0015c6c59a1b6cf55f4..4c2890c57c017bf2e9abd0a9c2838d6fefb93d17 100644
--- a/package.json
+++ b/package.json
@@ -24,7 +24,8 @@
 		"proB"
 	],
 	"activationEvents": [
-		"onLanguage:plaintext"
+		"onLanguage:B",
+		"onLanguage:EventB"
 	],
 	"main": "./client/out/extension",
 	"contributes": {
@@ -68,7 +69,24 @@
 					"description": "Option for stricter Checks"
 				}
 			}
-		}
+		},
+		"languages": [
+			{
+				"id": "B",
+				"extensions": [
+					".mch",
+					".ref",
+					".imp",
+					".def"
+				]
+			},
+			{
+				"id": "EventB",
+				"extensions": [
+					".sys"
+				]
+			}
+		]
 	},
 	"scripts": {
 		"vscode:prepublish": "npm run compile",
@@ -99,4 +117,4 @@
 		"vscode-api": "0.0.0",
 		"vscode-uri": "^2.1.1"
 	}
-}
+}
\ No newline at end of file