minor cleanup on example code