Update setup.py

Updated setup.py file for proper installation