Add colors to logging functions