diff --git a/risc-v_shell.tlv b/risc-v_shell.tlv
index d3a9792..4d63404 100644
--- a/risc-v_shell.tlv
+++ b/risc-v_shell.tlv
@@ -46,6 +46,9 @@
    
    // YOUR CODE HERE
    // ...
+   $next_pc[31:0] = $reset ? 0 : ($pc + 4);
+
+   $pc[31:0] = >>1$next_pc;
    
    
    // Assert these to end simulation (before Makerchip cycle limit).