simplify example in docs
fix code indentation