Minor tweak in output