Adding some comments onto the existing code