Use of ElementArrayStim in Code chunk

Coding
1

Hello,

I am trying to display arrays of a rectangle visual stimulus with 9 single rectangles each in a local experiment. Each rectangle is supposed to have a colour that was picked randomly during a previous part of the experiment (blue or red). It is supposed to be some sort of card game where the computer shows you the card colors if you win a lottery. I want to put up a summary screen at the end of my practice trials to show my participants all the cards with their assigned colors. Unfortunately I do not know how to make it work, I only get a blank screen right now. This is the code I figured out yet. I put it in the Begin routine tab:

fieldPos = (0.0, 0.0)
fieldSize = (400, 400)
nElements = 9
sizes = 15.0
xys = None 
colors = ('black')
colorSpace = 'rgb'
opacities = -1.0
depths = 0
fieldDepth = 0
oris = 0
sfs = 1.0
contrs = 1
phases = 0
elementTex = 'sin'
elementMask = 'gauss'
texRes = 48

element_array = visual.ElementArrayStim(
    win=win,
    fieldPos=fieldPos,
    fieldSize= fieldSize,
    nElements= nElements,
    sizes= sizes,
    xys=xys,
    colors= colors,
    colorSpace=colorSpace,
    opacities=opacities,
    depths=depths,
    fieldDepth=fieldDepth,
    oris=oris,
    sfs=sfs,
    contrs=contrs,
    phases=phases,
    elementTex=elementTex,
    elementMask=elementMask,
    texRes=texRes
)

element_array.draw()
win.flip()
event.waitKeys()
win.close()

what can I do to display the rectangles correctly and also get the colour information from my experiment data?