Add some imports
Add travis build badge
Fix #7