Update to the output. Bit nicer formatting.

Fixes #2