diff --git a/community/lms/doctype/exercise/exercise.py b/community/lms/doctype/exercise/exercise.py
index 66aaf35f..151b993a 100644
--- a/community/lms/doctype/exercise/exercise.py
+++ b/community/lms/doctype/exercise/exercise.py
@@ -43,6 +43,8 @@ class Exercise(Document):
         course = frappe.get_doc("LMS Course", self.course)
         batch = course.get_student_batch(user)
 
+        image = livecode_to_svg(None, code)
+
         doc = frappe.get_doc(
             doctype="Exercise Submission",
             exercise=self.name,
@@ -50,6 +52,7 @@ class Exercise(Document):
             course=self.course,
             lesson=self.lesson,
             batch=batch and batch.name,
+            image=image,
             solution=code)
         doc.insert(ignore_permissions=True)
         return doc
diff --git a/community/lms/doctype/exercise_submission/exercise_submission.json b/community/lms/doctype/exercise_submission/exercise_submission.json
index de7541ea..bd662b11 100644
--- a/community/lms/doctype/exercise_submission/exercise_submission.json
+++ b/community/lms/doctype/exercise_submission/exercise_submission.json
@@ -10,7 +10,8 @@
   "exercise_title",
   "course",
   "batch",
-  "lesson"
+  "lesson",
+  "image"
  ],
  "fields": [
   {
@@ -54,11 +55,17 @@
    "fieldtype": "Link",
    "label": "Lesson",
    "options": "Lesson"
+  },
+  {
+   "fieldname": "image",
+   "fieldtype": "Code",
+   "label": "Image",
+   "read_only": 1
   }
  ],
  "index_web_pages_for_search": 1,
  "links": [],
- "modified": "2021-05-20 13:30:16.349278",
+ "modified": "2021-05-21 11:28:45.833018",
  "modified_by": "Administrator",
  "module": "LMS",
  "name": "Exercise Submission",