fixes #2 - tests created

Reworked the gem so that the tests are done.