diff --git a/.vscode/settings.json b/.vscode/settings.json
index de288e1ea..6a4e11582 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -1,3 +1,7 @@
 {
-    "python.formatting.provider": "black"
-}
\ No newline at end of file
+  "python.testing.pytestArgs": ["../course/tests", "-rA"],
+  "python.testing.autoTestDiscoverOnSaveEnabled": true,
+  "python.testing.unittestEnabled": false,
+  "python.testing.nosetestsEnabled": false,
+  "python.testing.pytestEnabled": true
+}
diff --git a/set1/exercise1.py b/set1/exercise1.py
index 431db4017..3a049ed58 100644
--- a/set1/exercise1.py
+++ b/set1/exercise1.py
@@ -1,12 +1,14 @@
-"""Your very first python program!
+"""Your very first python program!"""
 
-TODO: write a python script that prints "Hello world!"
 
-Look on the course website for how to do this, and for how to run the tests.
+def hello():
+    """
+    TODO: write a python script that prints "Hello world!"
+    Look on the course website for how to do this, and for how to run the tests.
+    https://design-computing.github.io/md/week1
+    """
+    # 👇👇 Write your code here 👇👇
 
-https://design-computing.github.io/md/week1
+    print("Hello world!")
 
-"""
-# 👇👇 Write your code here 👇👇
-print("Hello world!")
-# 👆👆 Write your code here 👆👆
+    # 👆👆 Write your code here 👆👆