Add more tests

We should add more tests. While I like the idea of including tests in the same commit as adding the functionality under test, having users do TDD during a tutorial may be too much...

Add tests at the end as "bonus" content?