Made setup script compatible with python >= 3