fixed the color change at the end of the script