Adding better output support for gem