Remove unnecessary complication from code example