Added option for colorized output