Removed unused import os
Removed unused function