Slightly better example code in doc