improve color on update