Add travis tests
Remove main.py