Add setup.py for easy installation