Improved the output by a little bit