Added some comments for my code in the previous commit