diff --git a/src/app/webide/webide.component.ts b/src/app/webide/webide.component.ts
index f656f6d..a6abcfd 100644
--- a/src/app/webide/webide.component.ts
+++ b/src/app/webide/webide.component.ts
@@ -30,7 +30,9 @@ export class WebideComponent implements OnInit {
       this.code = event.data.content;
       this.language = event.data.language;
       this.files = event.data.files;
+
       if (event.data.command === 'write') { this.saveButtonState = 'SAVED'; }
+      if (event.data.command === 'reload') { this.requestCode(); }
     });
     this.requestCode();
   }