Added output bugfix