Added colors in unit test