Improve connection handling and exiting
Add support for colors