Update Gemfile
Add tests