Tidied up example in documentation