Minor modification in ScreenOutputGenerator