Add some comments for better reading of the code