Clean up example implementation