diff --git a/code.py b/code.py
index cfbc27724..287a42f99 100644
--- a/code.py
+++ b/code.py
@@ -1,6 +1,7 @@
 import random
 import pdb
 import sys
+import os
 
 
 class BaseNumberGenerator: