Added support for colored output