Shape specification for color update more robust.
Merge pull request #81 from tsherif/color-update-fix

Shape specification for color update more robust.