added color to the script