diff --git a/README.md b/README.md
index 5a7c609..9eb537d 100644
--- a/README.md
+++ b/README.md
@@ -1 +1,9 @@
-# comixify
\ No newline at end of file
+# comixify
+
+#TODO:
+- [ ] Prepare readme
+- [ ] Fix model loading to memory
+- [ ] Replace ffmpeg with some keyframes extractor
+- [ ] Prepare own ComixGAN
+- [ ] Prepare some more intelligent method for comic layout
+- [ ] Prepare some method for voice extraction
\ No newline at end of file
diff --git a/settings/settings.py b/settings/settings.py
index e165092..8700fef 100644
--- a/settings/settings.py
+++ b/settings/settings.py
@@ -25,7 +25,7 @@ SECRET_KEY = 'secret_123'
 # SECURITY WARNING: don't run with debug turned on in production!
 DEBUG = True
 
-ALLOWED_HOSTS = ['35.241.250.34', 'comixify.ii.pw.edu.pl']
+ALLOWED_HOSTS = ['35.241.250.34', 'comixify.ii.pw.edu.pl', 'localhost', '127.0.0.1']
 
 
 # Application definition
diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py
index c0b2fc9..76f183d 100644
--- a/style_transfer/style_transfer.py
+++ b/style_transfer/style_transfer.py
@@ -21,6 +21,7 @@ class StyleTransfer():
         style = kwargs.get("style", "Hayao")
         resize = kwargs.get("resize", 450)
 
+        # TODO: We should load model to memory right after deployment, not on each request.
         # load pretrained model
         model = Transformer()
         model.load_state_dict(torch.load(os.path.join("CartoonGAN/pretrained_model", style + "_net_G_float.pth")))