Add new prompt coloring code