diff --git a/json_tests.py b/json_tests.py
index d89aa078b5e6c0a59193c0d8ac059cb9526625b5..2a0463c6cc863bc39cab95284c55198705a2f5aa 100644
--- a/json_tests.py
+++ b/json_tests.py
@@ -3,7 +3,7 @@ import unittest
 
 import numpy as np
 
-from main import load_task, grid2FOL, FOL2grid, FOL2prolog, prolog2FOL_array, run_prolog_program, array_and_plot_grid
+from main import load_task, FOL2grid, FOL2prolog, prolog2FOL_array
 from task import Task
 
 
diff --git a/main.py b/main.py
index 469c6b1b07a4ce8469ec9f5f344c6c03399438ec..6095458436bd1246a4ad202d4955f4a8482d0dd1 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,14 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import subprocess
 
+tasks = {
+    "lines": "0b148d64.json",
+    "grids": "90f3ed37.json",
+    "pour": "d4f3cd78.json",
+    "cross": "e21d9049.json",
+    "stripes": "f8c80d96.json"
+}
+
 
 def run_prolog_program(program, curr_dir=""):
     # Construct the command to run SICStus Prolog