Edited the comments in the code