Update setup file
Fix for python3