Updating code sample not in line with current puzzler formulation