diff --git a/FIT5032-Assignment/Controllers/AppointmentsController.cs b/FIT5032-Assignment/Controllers/AppointmentsController.cs
index e4f6d1d..c4c3150 100644
--- a/FIT5032-Assignment/Controllers/AppointmentsController.cs
+++ b/FIT5032-Assignment/Controllers/AppointmentsController.cs
@@ -473,6 +473,7 @@ namespace FIT5032_Assignment.Controllers {
           file = fileName,
           status = 0,
         };
+        db.Images.Add(dbImage);
         // Change appointment status to 2
         appointment.status = 2;
         db.Entry(appointment).State = EntityState.Modified;
diff --git a/FIT5032-Assignment/Controllers/HomeController.cs b/FIT5032-Assignment/Controllers/HomeController.cs
index 04b3f9e..4710316 100644
--- a/FIT5032-Assignment/Controllers/HomeController.cs
+++ b/FIT5032-Assignment/Controllers/HomeController.cs
@@ -52,6 +52,7 @@ namespace FIT5032_Assignment.Controllers {
     public DbSet<Patients> Patients { get; set; }
     public DbSet<Doctors> Doctors { get; set; }
     public DbSet<Appointments> Appointments { get; set; }
+    public DbSet<Images> Images { get; set; }
   }
   public class HomeController : Controller {
     public static string GetMd5Hash(string input) {