Rewriting setup.py for easy installation