Clean up code example