Clean up example code