Add colorize for better visualisation

Use colorize gem todo all the stuff