fix up code samples