Fix colors in main