Added import os to make setup.py work