How to update color values when using a shader?

[prisoner849]

I slightly modified my example, so an individual box has its own color: https://jsfiddle.net/prisoner849/81rqxd20/