Made code sample more clear