added color to code