Added missing import os in setup.py