Improved logging and added support for colors