Add comment and reformat the code