Tidied up example a bit