Fixes #55 - it was the code that should have been there in the first place, but I forgot to `git add`...