Cleanup for setup.py
Added support for travis-ci