Added the appropriate comment and updated the indentation of the code