correcting the outpu