As I am slightly confused with this piece, please let me see this particular piece of functionality again and make changes if I made any mistakes by overlooking the code.