diff --git a/.vscode/settings.json b/.vscode/settings.json
new file mode 100644
index 0000000..3516cb9
--- /dev/null
+++ b/.vscode/settings.json
@@ -0,0 +1,3 @@
+{
+    "python.formatting.provider": "autopep8"
+}
\ No newline at end of file
diff --git a/code/behavior.py b/code/behavior.py
index 4009c28..b89a28a 100644
--- a/code/behavior.py
+++ b/code/behavior.py
@@ -1,20 +1,15 @@
 from pathlib import Path
-import os 
-import numpy as np
-from IPython import embed
-
 from pandas import read_csv
 
 
-
 class Behavior:
-    def __init__(self, datapath: str):
+    def __init__(self, datapath: str) -> None:
         csv_file = str(sorted(Path(datapath).glob('**/*.csv'))[0])
         self.dataframe = read_csv(csv_file, delimiter=',')
 
 
-
-def main(datapath:str):
+def main(datapath: str):
+    # behabvior is pandas dataframe with all the data
     behavior = Behavior(datapath)