diff --git a/Website/main.py b/Website/main.py
index 1b9d3b9..469c7bf 100644
--- a/Website/main.py
+++ b/Website/main.py
@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, jsonify, flash, redirect, url_for, get_flashed_messages, session, send_file, after_this_request, abort
+from flask import Flask, render_template, request, jsonify, flash, redirect, url_for, get_flashed_messages, session, send_file, after_this_request, abort, send_from_directory
 import os
 import json
 import atexit