Added missing 'import os' line