Removed 'import os' from test.py