Adding setup.py support