Tidy up example
Tidy up example