Update lib.rs

[Denys-Bushulyak]

Add a comment to highlight the code flow. I was a bit stuck on this example and thought it had an error. However, when I found the return keyword, I understood the true flow. I would like to add this comment to make the flow of reading smoother.