Added a note to the code in the comment section