I took a look at the source code but I'm still a bit confused about what's going on. Would it be possible to add some docstring for context and clarity? Also, it might be helpful to improve the print statement for better understanding.