Fix for issue #7

The changes currently loose output coloring support and they might not replicate previous behaviour 100%