Minor example code re-oganization for clarity