Improve prompt and colors in general