Fixed setup script to support python >= 3