Add unit test
Clean up Gemfile
Add .travis.yml