Import os module in setup.py