Slight changes to output