Clarify the style function signature

Very minor, and doesn't affect the behavior of the example, but this clarifies what the second arg to the style function is.