minor changes
fix code example