Update setup.py

Fixed lines 11 and 18