Add small output improvement