Made small changed to output