FSE: Color assignment - improved