Add tests

The new functionality should be tested. The project seems to use RSpec for behaviour-driven development/TDD.