Update output to make more sense