diff --git a/docs/FileHandler.puml b/docs/FileHandler.puml
index 8af56a5869..88e71240b7 100644
--- a/docs/FileHandler.puml
+++ b/docs/FileHandler.puml
@@ -2,16 +2,27 @@
 'https://plantuml.com/class-diagram
 
 class FileHandler {
-  +initFile(Records): void
-  +initFile(AppointmentRecord): void
-  +convertPatientToOutputText(Records): String
-  +convertPatientToOutputText(AppointmentRecord): String
-  +autosave(Records): void
-  +autosave(AppointmentRecord): void
-  +retrieveData(Records): void
-  +retrieveData(AppointmentRecord): void
-  +parseVisitInputString(visitString): Visit
-  +parseList(input): ArrayList<String>
+  -{static} Logger logger
+  -{static} String filePath
+  -{static} String appointmentFilePath
+  +{static} initFile(Records): void
+  +{static} initFile(AppointmentRecord): void
+  +{static} convertPatientToOutputText(Records): String
+  +{static} convertPatientToOutputText(AppointmentRecord): String
+  +{static} autosave(Records): void
+  +{static} autosave(AppointmentRecord): void
+  +{static} retrieveData(Records): void
+  +{static} retrieveData(AppointmentRecord): void
+  +{static} parseVisitInputString(visitString): Visit
+  +{static} parseList(input): ArrayList<String>
 }
 
+class Main {}
+class CommandHandler {}
+
+CommandHandler --> FileHandler: calls autosave
+Main --> FileHandler: calls initFile
+FileHandler --> FileHandler: initFile calls retrieveData
+FileHandler --> FileHandler: retrieveData calls parseVisitInputString and parseList
+
 @enduml
\ No newline at end of file
diff --git a/docs/FindVisit.puml b/docs/FindVisit.puml
index 1ab380d2a7..5a1a377e56 100644
--- a/docs/FindVisit.puml
+++ b/docs/FindVisit.puml
@@ -2,9 +2,19 @@
 'https://plantuml.com/class-diagram
 
 class FindVisit {
-  +findVisitByIc(nric, records): void
-  +findVisitByDiagnosis(Diagnosis, records): void
-  +findVisitByMedication(Medication, records): void
+  +{static} findVisitByIc(nric, records): void
+  +{static} findVisitByDiagnosis(Diagnosis, records): void
+  +{static} findVisitByMedication(Medication, records): void
 }
 
+class Patient {}
+class Visit {}
+class Main {}
+class Records {}
+
+Main --> FindVisit : calls >
+FindVisit --> Visit : searches >
+Records "1" *-- "many" Patient
+Patient *-- "many" Visit
+
 @enduml
\ No newline at end of file