Add print_color function
Refactor log functions to use print_color fn