Clean up example

Clean up example