Make code example valid Python