added color correction and fixed error in writing YML file