Clean up example more