Add tests and update Gemfile to include dependencies