Improve testing

- [x] add flake8 to travis
- [x] add sample data testing