Edited the comments for style