Clean up example more

Clean up example more