Updates to getColorSquare