Better solution now to the problem of the output