Tidied up example

Signed-off-by: Ollie <ollie@oliver-new.net>