Added colors to code