Slight changes to adding comments