Little improvement on example