Add tests

It'd be nice to have automatic tests of the basic functionality. Sadly, I don't think Travis is very good at terminal interaction. Perhaps with a bit of refactoring, it would be possible to set up some simple unit tests?

Any suggestions or comments are welcome.
