Added missing os import in setup.py