diff --git a/wally-pipelined/testgen/privileged/testgen-IE.py b/wally-pipelined/testgen/privileged/testgen-IE.py
index 67bdf67e5..ef8a524c2 100644
--- a/wally-pipelined/testgen/privileged/testgen-IE.py
+++ b/wally-pipelined/testgen/privileged/testgen-IE.py
@@ -3,6 +3,7 @@
 # testgen-IE.py
 #
 # ushakya@hmc.edu 31 March 2021
+# Modified: 4 April 2021
 #
 # Generate directed and random test vectors for RISC-V Design Validation.
 ##################################
@@ -21,67 +22,212 @@ from random import getrandbits
 
 def writeTrapHandlers(storecmd):
   global testnum
-  reg1 = 30
-  reg2 = 29
-  reg3 = 28
-  lines = "\n# Trap Handler: Timer Interupt\n"
-  lines += "_timer_trap_handler:\n"
+  [reg1, reg2, reg3] = [30, 29, 28]
+  [reg4, reg5] = [27, 26]
+  lines = "\n# Trap Handler: Machine Timer Interupt\n"
+  lines += "_timerM_trap_handler:\n"
   lines += "li x" + str(reg1) + ", MASK_XLEN(0x2A)\n"
-  lines += str(storecmd) + " x" + str(reg1) + ", " + str(wordsize*testnum) + "(x6)\n"
   lines += "la x" + str(reg2) + ", 0x2004000\n"
   lines += str(storecmd) + " x" + str(reg1) + ",  0(x" + str(reg2) + ")\n"
   lines += "csrrw x" + str(reg3) + ", mepc, x0\n"
   lines += "addi x"+ str(reg3) + ", x" + str(reg3) + ", MASK_XLEN(0x4)\n"
   lines += "mret\n"
 
+  lines += "\n# Trap Handler: Supervisor Timer Interupt\n"
+  lines += "_timerS_trap_handler:\n"
+  lines += "li x" + str(reg4) + ", MASK_XLEN(0x20)\n"
+  lines += "csrrc x0, mip, x" + str(reg4) + "\n"
+  lines += "csrrw x" + str(reg5) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg5) + ", x" + str(reg5) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: User Timer Interupt\n"
+  lines += "_timerU_trap_handler:\n"
+  lines += "li x" + str(reg4) + ", MASK_XLEN(0x10)\n"
+  lines += "csrrc x0, mip, x" + str(reg4) + "\n"
+  lines += "csrrw x" + str(reg5) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg5) + ", x" + str(reg5) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: Machine Software Interupt\n"
+  lines += "_softwareM_trap_handler:\n"
+  lines += "li x" + str(reg1) + ", MASK_XLEN(0x0)\n" # clear MSIP bit in CLINT
+  lines += "la x" + str(reg2) + ", 0x2000000\n"
+  lines += str(storecmd) + " x" + str(reg1) + ",  0(x" + str(reg2) + ")\n"
+  lines += "csrrw x" + str(reg3) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg3) + ", x" + str(reg3) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: Supervisor Software Interupt\n"
+  lines += "_softwareS_trap_handler:\n"
+  lines += "li x" + str(reg4) + ", MASK_XLEN(0x2)\n"
+  lines += "csrrc x0, mip, x" + str(reg4) + "\n"
+  lines += "csrrw x" + str(reg5) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg5) + ", x" + str(reg5) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: User Software Interupt\n"
+  lines += "_softwareU_trap_handler:\n"
+  lines += "li x" + str(reg4) + ", MASK_XLEN(0x1)\n"
+  lines += "csrrc x0, mip, x" + str(reg4) + "\n"
+  lines += "csrrw x" + str(reg5) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg5) + ", x" + str(reg5) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: Machine External Interupt\n"
+  lines += "_externalM_trap_handler:\n"
+  #lines += "li x" + str(reg1) + ", MASK_XLEN(0x0)\n" # clear MSIP bit in CLINT
+  #lines += "la x" + str(reg2) + ", 0x2000000\n"
+  #lines += str(storecmd) + " x" + str(reg1) + ",  0(x" + str(reg2) + ")\n"
+  lines += "csrrw x" + str(reg3) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg3) + ", x" + str(reg3) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: Supervisor External Interupt\n"
+  lines += "_externalS_trap_handler:\n"
+  lines += "li x" + str(reg4) + ", MASK_XLEN(0x200)\n"
+  lines += "csrrc x0, mip, x" + str(reg4) + "\n"
+  lines += "csrrw x" + str(reg5) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg5) + ", x" + str(reg5) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
+  lines += "\n# Trap Handler: User External Interupt\n"
+  lines += "_externalU_trap_handler:\n"
+  lines += "li x" + str(reg4) + ", MASK_XLEN(0x100)\n"
+  lines += "csrrc x0, mip, x" + str(reg4) + "\n"
+  lines += "csrrw x" + str(reg5) + ", mepc, x0\n"
+  lines += "addi x"+ str(reg5) + ", x" + str(reg5) + ", MASK_XLEN(0x4)\n"
+  lines += "mret\n"
+
   f.write(lines)
 
-def writeVector(a, xlen, storecmd):
+def getInteruptEnableValues():
+  if test == "timerM":
+    mstatusE = 0x8
+    mieE = 0x80
+  elif test == "timerS":
+    mstatusE = 0x2
+    mieE = 0x20
+  elif test == "timerU":
+    mstatusE = 0x1
+    mieE = 0x10
+  elif test == "softwareM":
+    mstatusE = 0x8
+    mieE = 0x8
+  elif test == "softwareS":
+    mstatusE = 0x2
+    mieE = 0x2
+  elif test == "softwareU":
+    mstatusE = 0x1
+    mieE = 0x1
+  elif test == "externalM":
+    mstatusE = 0x8
+    mieE = 0x800
+  elif test == "externalS":
+    mstatusE = 0x2
+    mieE = 0x200
+  elif test == "externalU":
+    mstatusE = 0x1
+    mieE = 0x100
+  return [mstatusE, mieE]
+
+def getMcause():
+  b = 1 << (xlen-1)
+  if test == "timerM":
+    b = b + 0x7
+  elif test == "timerS":
+    b = b + 0x5
+  elif test == "timerU":
+    b = b + 0x4
+  elif test == "softwareM":
+    b = b + 0x3
+  elif test == "softwareS":
+    b = b + 0x1
+  elif test == "softwareU":
+    b = b
+  elif test == "externalM":
+    b = b + 0xB
+  elif test == "externalS":
+    b = b + 0x9
+  elif test == "externalU":
+    b = b + 0x8
+  return b
+
+def writeVectors(a, xlen, storecmd):
   global testnum
 
   [reg1, reg2, reg3] = [1, 2, 3]
   [reg5, reg8] = [5, 8]
   [reg9, reg10, reg11, reg12] = [9, 10, 11, 12]
 
-  lines = "\n# Testcase 0: Timer Interupt\n"
-
-  # Page 6 of unpriviledged spec
-  # For both CSRRS and CSRRC, if rs1=x0, then the instruction will not write to the CSR at all, and so shall not cause any of the side effects
+  lines = f"\n# Testcase {testnum}: {test} Interupt\n"
   
   # mcause code
-  b = 1 << (xlen-1)
-  b = b + 0x7
-  expected = b
-  lines = lines + "li x" + str(reg1) + ", MASK_XLEN(" + formatstr.format(b) + ")\n"
+  expected = getMcause()
+  lines = lines + "li x" + str(reg1) + ", MASK_XLEN(" + formatstr.format(expected) + ")\n"
 
   if (testnum == 0): expected = 0
 
+  [mstatusE, mieE] = getInteruptEnableValues()
   # set interupt enable bit in mstatus
-  lines += "li x" + str(reg3) + ", MASK_XLEN(0x8)\n"
+  lines += "li x" + str(reg3) + ", MASK_XLEN(" + str(mstatusE) + ")\n"
   lines += "csrrs x0, mstatus, x" + str(reg3) + "\n"
 
-  # set machine timer interupt enable bit in mie
-  lines += "li x" + str(reg9) + ", MASK_XLEN(0x80)\n"
+  # set timer interupt enable bit in mie
+  lines += "li x" + str(reg9) + ", MASK_XLEN(" + str(mieE) + ")\n"
   lines += "csrrs x0, mie, x" + str(reg3) + "\n"
 
-  # Save and set trap handler address for machine mode timer interrupt
-  lines += "la x" + str(reg5) + ", _timer_trap_handler\n"
+  # Save and set trap handler address for interrupt
+  lines += "la x" + str(reg5) + ", _" + test + "_trap_handler\n"
   
   # save orignal mtvec address
   lines += "csrrw x" + str(reg12) + ", mtvec, x" + str(reg5) + "\n"
   
   # cause timer interupt
-  #if (testnum == 0):
-  lines += "li x" + str(reg8) + ", MASK_XLEN(0)\n"
-  lines += str(storecmd) + " x" + str(reg8) + ", " + str(wordsize*testnum)+ "(x6)\n"
+  if test == "timerM":
+    lines += "li x" + str(reg8) + ", MASK_XLEN(0)\n"
+    lines += str(storecmd) + " x" + str(reg8) + ", " + str(wordsize*testnum)+ "(x6)\n"
+    
+    lines += "la x" + str(reg8) + ", 0x2004000\n"
+
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0)\n"
+
+    # save old value of mtimecmp and then set mtimecmp to zero
+    lines += "lw x" + str(reg11) + ", 0(x" + str(reg8) + ")\n"
+    lines += str(storecmd) + " x" + str(reg3) + ",  0(x" + str(reg8) + ")\n"
+  elif test == "timerS":
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x20)\n"
+    lines += "csrrs x0, mip, x" + str(reg3) + "\n"
+  elif test == "timerU":
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x10)\n"
+    lines += "csrrs x0, mip, x" + str(reg3) + "\n"
   
-  lines += "la x" + str(reg8) + ", 0x2004000\n"
+  # cause software interupt
+  if test == "softwareM":
+    lines += "la x" + str(reg8) + ", 0x2000000\n" # Write to the MSIP bit in CLINT
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x1)\n"
+    lines += str(storecmd) + " x" + str(reg3) + ",  0(x" + str(reg8) + ")\n"
+  elif test == "softwareS":
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x2)\n"
+    lines += "csrrs x0, mip, x" + str(reg3) + "\n"
+  elif test == "softwareU":
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x1)\n"
+    lines += "csrrs x0, mip, x" + str(reg3) + "\n"
+  
+  # cause external interupt
+  # Not sure how to cause an external machine interupt yet
+  # will writing to PLIC just cause it? (where is the ExtIntM located in PLIC)
+  #if test == "externalM":
+    #lines += "la x" + str(reg8) + ", 0x2000000\n" # Write to the MSIP bit in CLINT
+    #lines += "li x" + str(reg3) + ", MASK_XLEN(0x1)\n"
+    #lines += str(storecmd) + " x" + str(reg3) + ",  0(x" + str(reg8) + ")\n"
+  if test == "externalS":
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x200)\n"
+    lines += "csrrs x0, mip, x" + str(reg3) + "\n"
+  elif test == "externalU":
+    lines += "li x" + str(reg3) + ", MASK_XLEN(0x100)\n"
+    lines += "csrrs x0, mip, x" + str(reg3) + "\n"
 
-  lines += "li x" + str(reg3) + ", MASK_XLEN(0)\n"
-
-  # save old value of mtimecmp and then set mtimecmp to zero
-  lines += "lw x" + str(reg11) + ", 0(x" + str(reg8) + ")\n"
-  lines += str(storecmd) + " x" + str(reg3) + ",  0(x" + str(reg8) + ")\n"
   #lines += "wfi\n" # wait for interupt to be taken
   lines += "nop\nnop\n"
 
@@ -105,7 +251,7 @@ def writeVector(a, xlen, storecmd):
 ##################################
 
 # change these to suite your tests
-tests = ["timer"]
+tests = ["timerM", "timerS", "timerU", "softwareM", "softwareS", "softwareU"]
 author = "ushakya@hmc.edu"
 xlens = [64, 32]
 numrand = 100;
@@ -149,8 +295,7 @@ for xlen in xlens:
     # print directed and random test vectors
     for i in range(0,numrand):
       a = getrandbits(xlen)
-      b = getrandbits(xlen)
-      writeVector(a, xlen, storecmd)
+      writeVectors(a, xlen, storecmd)
 
     writeTrapHandlers(storecmd)
 
@@ -165,231 +310,3 @@ for xlen in xlens:
     f.write(lines)
     f.close()
     r.close()
-
-"""
-#!/usr/bin/python3
-##################################
-# testgen-IE.py
-#
-# ushakya@hmc.edu 24 Mar 2021
-#
-# Generate tests for mie CSR for RISC-V Design Validation.
-##################################
-
-##################################
-# libraries
-##################################
-from datetime import datetime
-from random import randint 
-from random import seed
-from enum import Enum
-from random import getrandbits
-
-##################################
-# functions
-##################################
-
-def randRegs():
-  reg1 = randint(1,30)
-  reg2 = randint(1,30)
-  reg3 = randint(1,30)
-  if (reg1 == 6 or reg2 == 6 or reg3 == 6 or reg1 == reg2):
-    return randRegs()
-  else:
-      return reg1, reg2, reg3
-
-def writeVectors(storecmd):  
-  global testnum
-  reg1, reg2, reg3 = randRegs()
-
-  # Set interupt enable bit in mstatus
-  lines = ""
-  li x2, 0x8
-  csrrs x3, mstatus, x2
-  ""
-
-  f.write(lines)
-
- # Save and set trap handler address for machine mode timer interrupt
-  lines += ""
-  la x1, _timer_trap_handler
-  csrrw x31, mtvec, x1
-  ""
-  f.write(lines)
-
-  # Machine Mode Timer Interrupt (when interupt is enabled)
-  # is this not working because mtimecmp isn't implemented????
-  write(f""
-    li x2, 0x0
-
-    li x4, 0x80
-    csrrs x0, mie, x4
-
-    {storecmd} x2, {str(wordsize*testnum)}(x6)
-
-    la x2, 0x2004000
-    
-    li x3, 0x0
-    lw x5, 0(x2)
-    sd x3, 0(x2)
-    wfi
-  "", storecmd, True, 4, "m")
-
-  # Supervisor Timer Interrupt
-  # user timer interupt
-
-  # Machine mode external interrupt (hasn't been connected yet)
-
-  # User external interrupt True, 8
-  # Supervisor external interrupt True, 9
-
-  # Save and set trap handler address for machine mode software interrupt
- # lines = ""
- # la x1, _interupt_trap_handler
- # csrrw x31, mtvec, x1
- # ""
- # f.write(lines)
- 
-  # Machine Mode software interupt (write to the CLINT)
-  #write(f""
-  #  li x6, 0x0
-#
-#    li x4, 0x8
-#    csrrs x0, mie, x4
-#    
-#    li x3, 0x1
-#    lw x4, clint
-#    or x3, x4, x3
-#    {storecmd} x3, clint
-#  "", storecmd, True, 3, "m")
-
-  # supervisor mode software interupt
-  # user mode software interupt
-  
-  # timer interupt trap handler
-  lines = f""
-  _timer_trap_handler:
-  li x2, 0x2A
-  {storecmd} x2, {str(wordsize*testnum)}(x6)
-  la x3, 0x2004000
-  {storecmd} x2, 0(x3)
-  mret
-  ""
-
-  # software interupt trap handler
-  #lines += f""
-  #_interupt_trap_handler:
-  #li x6, 0x2A
-  #li x3, 0x0
-  #lw x4, clint
-  #xor x3, x4, x3
-  #{storecmd} x3, 0(clint)
- #mret
-  #""
-  lines += storecmd + " x" + str(reg3) + ", " + str(wordsize*testnum) + "(x6)"
-
-  f.write(lines)
-
-def write(lines, storecmd, interrupt, code, mode = "m"):
-  global testnum
-
-  # generate expected interrupt code
-  expected = 0
-  #(0 if not interrupt else (2**31 if xlen == 32 else 2**63)) + code
-  # go back and fix expected
-
-  lines = f""
-    # Testcase {testnum}
-    li x31, 0
-    {lines}
-
-    {storecmd} x31, {str(wordsize*testnum)}(x6)
-    # RVTEST_IO_ASSERT_GPR_EQ(x0, 0, {formatstr.format(expected)})
-  ""
-
-  #if mode == "s":
-    # go to supervisor mode
-  #elif mode == "u":
-    # go to user mode
-
-  f.write(lines)
-
-  if (xlen == 32):
-    line = formatrefstr.format(expected)+""
-  else:
-    line = formatrefstr.format(expected % 2**32)+"" + formatrefstr.format(expected >> 32) + ""
-  r.write(line)
-  testnum = testnum+1
-
-##################################
-# main body
-##################################
-
-# name: (interrupt?, code)
-# tests = {
-#  'User software interrupt': (1, '0'),
-#  'Supervisor software interrupt': (1, '1'),
-#  'Machine software interrupt': (1, '3'),
-#  'User timer interrupt': (1, '4'),
-#  'Supervisor timer interrupt': (1, '5'),
-#  'Machine timer interrupt': (1, '7'),
-#  'User external interrupt': (1, '8'),
-#  'Supervisor external interrupt': (1, '9'),
-#  'Machine external interrupt': (1, '11'),
-# }
-author = "Udeema Shakya (ushakya@hmc.edu)"
-xlens = [64, 32]
-numrand = 60;
-
-# setup
-seed(0xC395DDEB9173AD42) # make tests reproducible
-
-# generate files for each test
-for xlen in xlens:
-  formatstrlen = str(int(xlen/4))
-  formatstr = "0x{:0" + formatstrlen + "x}" # format as xlen-bit hexadecimal number
-  formatrefstr = "{:08x}" # format as xlen-bit hexadecimal number with no leading 0x
-  if (xlen == 32):
-    storecmd = "sw"
-    wordsize = 4
-  else:
-    storecmd = "sd"
-    wordsize = 8
-
-  imperaspath = f"../../../imperas-riscv-tests/riscv-test-suite/rv{xlen}p/""
-  basename = "WALLY-IE"
-  fname = imperaspath + "src/" + basename + ".S"
-  refname = imperaspath + "references/" + basename + ".reference_output"
-  testnum = 0
-
-  # print custom header part
-  f = open(fname, "w")
-  r = open(refname, "w")
-  line = "///////////////////////////////////////////"
-  f.write(line)
-  lines="// "+fname+ "// " + author + ""
-  f.write(lines)
-  line ="// Created " + str(datetime.now()) 
-  f.write(line)
-
-  # insert generic header
-  h = open("../testgen_header.S", "r")
-  for line in h:  
-    f.write(line)
-
-  # print directed and random test vectors
-  writeVectors(storecmd)
-
-
-  # print footer
-  h = open("../testgen_footer.S", "r")
-  for line in h:  
-    f.write(line)
-
-  # Finish
-  lines = ".fill " + str(testnum) + ", " + str(wordsize) + ", -1"
-  lines = lines + "RV_COMPLIANCE_DATA_END" 
-   f.write(lines)
-  f.close()
-  r.close()
-  """