Clean up README code example