Cleaning up puts, making output prettier

We have colors now! :)