Fix coloring to prompt. Add exit color