Small changes to output